path: "tensorflow.experimental.dtensor.Layout"
tf_class {
  is_instance: "<class \'tensorflow.dtensor.python.layout.Layout\'>"
  is_instance: "<class \'tensorflow.python._pywrap_dtensor_device.Layout\'>"
  is_instance: "<class \'pybind11_builtins.pybind11_object\'>"
  member {
    name: "mesh"
    mtype: "<type \'property\'>"
  }
  member {
    name: "rank"
    mtype: "<type \'property\'>"
  }
  member {
    name: "shape"
    mtype: "<type \'property\'>"
  }
  member {
    name: "sharding_specs"
    mtype: "<type \'property\'>"
  }
  member {
    name: "type"
    mtype: "<type \'property\'>"
  }
  member_method {
    name: "__init__"
    argspec: "args=[\'self\', \'sharding_specs\', \'mesh\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "as_proto"
    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "batch_sharded"
    argspec: "args=[\'cls\', \'mesh\', \'batch_dim\', \'rank\', \'axis\'], varargs=None, keywords=None, defaults=[\'0\'], "
  }
  member_method {
    name: "delete"
    argspec: "args=[\'self\', \'dims\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "from_device"
    argspec: "args=[\'cls\', \'device\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "from_proto"
    argspec: "args=[\'cls\', \'layout_proto\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "from_single_device_mesh"
    argspec: "args=[\'cls\', \'mesh\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "from_string"
    argspec: "args=[\'cls\', \'layout_str\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "inner_sharded"
    argspec: "args=[\'cls\', \'mesh\', \'inner_dim\', \'rank\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "is_batch_parallel"
    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "is_fully_replicated"
    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "is_single_device"
    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "num_shards"
    argspec: "args=[\'self\', \'idx\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "offset_to_shard"
    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "offset_tuple_to_global_index"
    argspec: "args=[\'self\', \'offset_tuple\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "replicated"
    argspec: "args=[\'cls\', \'mesh\', \'rank\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "to_parted"
    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "to_string"
    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
  }
}
